import os
username = os.getenv("USER")

SENSOR_NAME_LIST = ["LHip", "RHip", "LThigh", "RThigh", "LKnee", "RKnee", 
                    "LShank", "RShank", "LHeel", "RHeel", "LToe", "RToe",
                    "Back", "Spine", "LShoulder", "RShoulder", "LBicep", "RBicep", 
                    "LElbow", "RElbow", "LForearm", "RForearm", "LHand", "RHand"]

SENSOR_NAME_MAPPER = {
    "LHip": "leftHip", 
    "RHip": "rightHip", 
    "LThigh": "leftThigh",
    "RThigh": "rightThigh",
    "LKnee": "leftKnee", 
    "RKnee": "rightKnee", 
    "LShank": "leftShank",
    "RShank": "rightShank",
    "LHeel": "leftFoot",
    "RHeel": "rightFoot",
    "LToe": "leftToeBase",
    "RToe": "rightToeBase",
    "Back": "lspine",
    "Spine": "uspine",
    "LShoulder": "leftShoulder",
    "RShoulder": "rightShoulder",
    "LBicep": "leftBicep",
    "RBicep": "rightBicep",
    "LElbow": "leftElbow",
    "RElbow": "rightElbow",
    "LForearm": "leftForeArm",
    "RForearm": "rightForeArm",
    "LHand": "leftHand",
    "RHand": "rightHand",
}


# Auxiliary directories
SMPL_MODEL_DIR = "you-smpl-model-dir"       # Download from https://smpl.is.tue.mpg.de/
SMPLX_MODEL_DIR = "your-smplx-model-dir"    # Download from https://smpl-x.is.tue.mpg.de/
SMPLX2SMPL_PTH = "your-smplx2smpl-dir"      # Download from https://smpl-x.is.tue.mpg.de/download.php

# 
DATA_BASE_DIR = "where-you-downloaded-data" 

### NOTE: The below is regarding the preprocessing, don't need to adjust the lines.
SEQUENCE_NAME = ""
PREPROC_BASE_DIR = "/media/soyongs/T9/projects"
ARIA_CAMERA_NAME = "aria01"     # Aria01
SLAM_CAMERA_NAME = "mobile"        # GoPro camera used for SLAM
EXO_CAMERA_NAMES = ["cam01", "cam02", "cam03", "cam04", "cam05", "cam06"]

# Framerate
ARIA_FPS = 30
GOPRO_FPS = 60
TARGET_FPS = 30

RAW_DATA_DIR = f"{PREPROC_BASE_DIR}/HumanEnv/raw_data/"
COLMAP_DIR = f"{PREPROC_BASE_DIR}/HumanEnv/colmap/"
EXTRACT_IMAGE_DIR = f"{PREPROC_BASE_DIR}/extracted"
PROCESSED_IMAGE_DIR = f"{PREPROC_BASE_DIR}/images"
BBOX_DIR = f"{PREPROC_BASE_DIR}/poses/bbox"
VITPOSE_RESULTS_DIR = f"{PREPROC_BASE_DIR}/poses/vitpose"
DENSE_VITPOSE_RESULTS_DIR = f"{PREPROC_BASE_DIR}/poses/dense_vitpose"
PROCESSED_SENSOR_DIR = f"{PREPROC_BASE_DIR}/contact"

# Execution directories
SAMURAI_DIR = ""
VITPOSE_DIR = ""
DENSE_VITPOSE_DIR = ""
SMPLIFYX_DIR = ""

# Checkpoints
YOLO_MODEL_CKPT = ""
SAMURAI_MODEL_CKPT = ""